1010-总持续时间可被 60 整除的歌曲

Raphael Liu Lv10

在歌曲列表中,第 i 首歌曲的持续时间为 time[i] 秒。

返回其总持续时间(以秒为单位)可被 60 整除的歌曲对的数量。形式上,我们希望下标数字 ij 满足 i < j 且有
(time[i] + time[j]) % 60 == 0

示例 1:

**输入:** time = [30,20,150,100,40]
**输出:** 3
**解释:** 这三对的总持续时间可被 60 整除:
(time[0] = 30, time[2] = 150): 总持续时间 180
(time[1] = 20, time[3] = 100): 总持续时间 120
(time[1] = 20, time[4] = 40): 总持续时间 60

示例 2:

**输入:** time = [60,60,60]
**输出:** 3
**解释:** 所有三对的总持续时间都是 120,可以被 60 整除。

提示:

  • 1 <= time.length <= 6 * 104
  • 1 <= time[i] <= 500

方法一:组合数学

思路

需要返回其总持续时间(以秒为单位)可被 60 整除的歌曲对的数量,因此,每首歌曲对结果的影响因素是它的持续时间除以 60 后的余数。可以用一个长度为 60 的数组 cnt,用来表示余数出现的次数。然后分情况统计歌曲对:

  1. 余数为 0 的歌曲。他们需要与余数为 0 的歌曲组成对,但不能与自己组成对。歌曲对的数量为 cnt}[0] \times (\textit{cnt}[0]-1)/2。
  2. 余数为 30 的歌曲。他们需要与余数为 30 的歌曲组成对,但不能与自己组成对。歌曲对的数量为 cnt}[30] \times (\textit{cnt}[30]-1)/2。
  3. 余数为 i, i\in[1,29] 的歌曲。他们需要与余数为 60-i 的歌曲组成对。歌曲对的数量为 \sum_{i=1}^{29}\textit{cnt}[i] \times \textit{cnt}[60-i]。
  4. 余数为 i, i\in[31,59] 的歌曲。已经在上一部分组对过,不需要重复计算。

把这几部分求和,就可以得到最后的对数。

代码

[sol1-Python3]
1
2
3
4
5
6
7
8
9
10
class Solution:
def numPairsDivisibleBy60(self, time: List[int]) -> int:
cnt = [0] * 60
for t in time:
cnt[t % 60] += 1
res = 0
for i in range(1, 30):
res += cnt[i] * cnt[60 - i]
res += cnt[0] * (cnt[0] - 1) // 2 + cnt[30] * (cnt[30] - 1) // 2
return res
[sol1-Java]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution {
public int numPairsDivisibleBy60(int[] time) {
int[] cnt = new int[60];
for (int t : time) {
cnt[t % 60]++;
}
long res = 0;
for (int i = 1; i < 30; i++) {
res += cnt[i] * cnt[60 - i];
}
res += (long) cnt[0] * (cnt[0] - 1) / 2 + (long) cnt[30] * (cnt[30] - 1) / 2;
return (int) res;
}
}
[sol1-C#]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
public class Solution {
public int NumPairsDivisibleBy60(int[] time) {
int[] cnt = new int[60];
foreach (int t in time) {
cnt[t % 60]++;
}
long res = 0;
for (int i = 1; i < 30; i++) {
res += cnt[i] * cnt[60 - i];
}
res += (long) cnt[0] * (cnt[0] - 1) / 2 + (long) cnt[30] * (cnt[30] - 1) / 2;
return (int) res;
}
}
[sol1-C++]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution {
public:
int numPairsDivisibleBy60(vector<int>& time) {
vector<int> cnt(60);
for (int t : time) {
cnt[t % 60]++;
}
long long res = 0;
for (int i = 1; i < 30; i++) {
res += cnt[i] * cnt[60 - i];
}
res += (long long)cnt[0] * (cnt[0] - 1) / 2 + (long long)cnt[30] * (cnt[30] - 1) / 2;
return (int)res;
}
};
[sol1-Go]
1
2
3
4
5
6
7
8
9
10
11
12
func numPairsDivisibleBy60(time []int) int {
cnt := make([]int, 60)
for _, t := range time {
cnt[t % 60]++
}
var res int
for i := 1; i < 30; i++ {
res += cnt[i] * cnt[60 - i]
}
res += cnt[0] * (cnt[0] - 1) / 2 + cnt[30] * (cnt[30] - 1) / 2
return res
}
[sol1-JavaScript]
1
2
3
4
5
6
7
8
9
10
11
12
var numPairsDivisibleBy60 = function(time) {
const cnt = new Array(60).fill(0);
for (let t of time) {
cnt[t % 60]++;
}
let res = 0;
for (let i = 1; i < 30; i++) {
res += cnt[i] * cnt[60 - i];
}
res += cnt[0] * (cnt[0] - 1) / 2 + cnt[30] * (cnt[30] - 1) / 2;
return res;
}
[sol1-C]
1
2
3
4
5
6
7
8
9
10
11
12
13
int numPairsDivisibleBy60(int* time, int timeSize) {
int cnt[60];
memset(cnt, 0, sizeof(cnt));
for (int i = 0; i < timeSize; i++) {
cnt[time[i] % 60]++;
}
long long res = 0;
for (int i = 1; i < 30; i++) {
res += cnt[i] * cnt[60 - i];
}
res += (long long)cnt[0] * (cnt[0] - 1) / 2 + (long long)cnt[30] * (cnt[30] - 1) / 2;
return (int)res;
}

复杂度分析

  • 时间复杂度:O(n),其中 n 是 time 的长度,需要遍历 time 一遍。

  • 空间复杂度:O(1),需要长度为 60 的数组。

 Comments
On this page
1010-总持续时间可被 60 整除的歌曲